import unittest
from test_cases import test_demo1, test_demo2

# TODO TestLoader第一种用法
'''
# 创建一个加载器对象, 加载测试类
loader = unittest.TestLoader()
unittest.defaultTestLoader
# unittest.defaultTestLoader  = unittest.TestLoader()
# 参数: 测试类
loader1 = loader.loadTestsFromTestCase(test_demo1.Demo1TestCase)
loader2 = loader.loadTestsFromTestCase(test_demo2.Demo2TestCase)
'''
# TODO TestLoader第二种用法
loader1 = unittest.defaultTestLoader.loadTestsFromTestCase(test_demo1.Demo1TestCase)
loader2 = unittest.defaultTestLoader.loadTestsFromTestCase(test_demo2.Demo2TestCase)

# 创建一个测试套件, 收集测试类
suite = unittest.TestSuite()
# addTest()
# suite.addTest(loader1)
# suite.addTest(loader2)
# addTests()
suite.addTests([loader1, loader2])

# 创建执行器,   执行测试类
runner = unittest.TextTestRunner()
runner.run(suite)
